package JZ_Offer.dataStructure.tree;

import base.TreeNode;

import java.util.Stack;

/**
 * 二叉搜索树的第k个结点
 * <p>
 * 给定一棵结点数为n 二叉搜索树，请找出其中的第 k 小的TreeNode结点值。
 * 1.返回第k小的节点值即可
 * 2.不能查找的情况，如二叉树为空，则返回-1，或者k大于n等等，也返回-1
 * 3.保证n个节点的值不一样
 * <p>
 * 例如，（5，3，7，2，4，6，8）中，按结点数值大小顺序第三小结点的值为4。
 */
public class JZ54_二叉搜索树的第k个结点 {

    private static TreeNode KthNode(TreeNode pRoot, int k) {
        if (k == 0) {
            return null;
        }

        Stack<TreeNode> stack = new Stack<>();
        recursive(pRoot, k, stack);
        if (stack.size() < k) {
            return null;
        }
        return stack.get(k - 1);
    }

    private static void recursive(TreeNode node, int k, Stack<TreeNode> stack) {
        if (node == null) {
            return;
        }

        // 存储临时出栈的节点
        Stack<TreeNode> tempStack = new Stack<>();
        // 找位置：依次比较当前节点值和栈里已经有的值 小于就出栈
        for (int i = stack.size(); i > 0 && node.val < stack.peek().val; i--) {
            tempStack.push(stack.pop());
        }

        // 入栈
        stack.push(node);

        // 将出栈的数据再放回来
        for (int i = tempStack.size(); i > 0 && stack.size() < k; i--) {
            stack.push(tempStack.pop());
        }

        // 递归
        recursive(node.left, k, stack);
        recursive(node.right, k, stack);
    }

    public static void main(String[] args) {
        TreeNode a = new TreeNode(8);
        TreeNode b = new TreeNode(6);
        TreeNode c = new TreeNode(10);
        TreeNode d = new TreeNode(5);
        TreeNode e = new TreeNode(7);
        TreeNode f = new TreeNode(9);
        TreeNode g = new TreeNode(11);

        a.left = b;
        a.right = c;
        b.left = d;
        b.right = e;
        c.left = f;
        c.right = g;

        TreeNode node = KthNode(a, 1);
        System.out.println();
    }

}
